import torch

x = torch.rand(4, 5)
torch.save(x.to(torch.device('cpu')), "myTensor.pth")

y = torch.load("myTensor.pth")
print(y)
